@echo off
chcp 65001 >nul
title 安装依赖

echo ============================================
echo   安装 Segment Anything 工具依赖
echo ============================================
echo.

REM 检查 Python 是否安装
python --version >nul 2>&1
if errorlevel 1 (
    echo [错误] 未检测到 Python！
    echo 请先安装 Python 3.8 或更高版本
    echo 下载地址: https://www.python.org/downloads/
    pause
    exit /b 1
)

echo [信息] 检测到 Python 版本:
python --version
echo.

echo [信息] 正在安装依赖库...
echo 这可能需要几分钟时间，请耐心等待...
echo.

REM 升级 pip
echo [步骤 1/3] 升级 pip...
python -m pip install --upgrade pip

if errorlevel 1 (
    echo [警告] pip 升级失败，继续安装...
)

echo.
echo [步骤 2/3] 安装 PyTorch...
echo 正在安装 CPU 版本的 PyTorch（如需 GPU 加速，请手动安装）
python -m pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu

if errorlevel 1 (
    echo [错误] PyTorch 安装失败！
    echo 请尝试手动安装: pip install torch torchvision
    pause
    exit /b 1
)

echo.
echo [步骤 3/3] 安装其他依赖...
python -m pip install -r requirements.txt

if errorlevel 1 (
    echo [错误] 依赖安装失败！
    pause
    exit /b 1
)

echo.
echo ============================================
echo   ✓ 安装完成！
echo ============================================
echo.
echo 接下来您可以:
echo 1. 运行 download_models.py 下载 SAM 模型
echo 2. 运行 sam_to_psd.py 启动工具
echo 或者直接双击【启动工具.bat】
echo.
echo 注意: 如果您有 NVIDIA GPU，建议安装 GPU 版本的 PyTorch
echo 请访问 https://pytorch.org 获取安装命令
echo.
pause

